package org.yzhzc.friend.service.question.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.github.pagehelper.PageHelper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Service;
import org.yzhzc.common.core.constants.Constants;
import org.yzhzc.common.core.domain.TableDataInfo;
import org.yzhzc.friend.domain.question.dto.QuestionQueryDTO;
import org.yzhzc.friend.domain.question.entity.Question;
import org.yzhzc.friend.domain.question.es.QuestionES;
import org.yzhzc.friend.domain.question.vo.QuestionDetailVO;
import org.yzhzc.friend.domain.question.vo.QuestionVO;
import org.yzhzc.friend.manager.QuestionCacheManager;
import org.yzhzc.friend.mapper.question.QuestionMapper;
import org.yzhzc.friend.mapper.user.UserSubmitMapper;
import org.yzhzc.friend.repository.question.QuestionRepository;
import org.yzhzc.friend.service.question.IQuestionService;

import java.util.ArrayList;
import java.util.List;

/**
 * 题目服务实现类。
 * 负责处理与题目相关的业务逻辑，包括题目列表查询、热门题目查询、题目详情查询以及题目导航等。
 */
@Slf4j
@Service
public class QuestionServiceImpl implements IQuestionService {

    @Autowired
    private QuestionRepository questionRepository; // Elasticsearch 题目数据访问接口

    @Autowired
    private QuestionMapper questionMapper; // 数据库题目数据访问接口

    @Autowired
    private UserSubmitMapper userSubmitMapper; // 用户提交记录数据访问接口

    @Autowired
    private QuestionCacheManager questionCacheManager; // 题目缓存管理类

    /**
     * 查询题目列表。
     *
     * @param questionQueryDTO 题目查询条件
     * @return 分页数据
     */
    @Override
    public TableDataInfo list(QuestionQueryDTO questionQueryDTO) {
        // 检查 Elasticsearch 中是否有数据，如果没有则刷新数据
        long count = questionRepository.count();
        if (count <= 0) {
            refreshQuestion();
        }

        // 构建分页和排序条件
        Sort sort = Sort.by(Sort.Direction.DESC, "createTime");
        Pageable pageable = PageRequest.of(questionQueryDTO.getPageNum() - 1, questionQueryDTO.getPageSize(), sort);

        // 根据查询条件执行不同的查询
        Integer difficulty = questionQueryDTO.getDifficulty();
        String keyword = questionQueryDTO.getKeyword();
        Page<QuestionES> questionESPage;
        if (difficulty == null && StrUtil.isEmpty(keyword)) {
            // 查询所有题目
            questionESPage = questionRepository.findAll(pageable);
        } else if (StrUtil.isEmpty(keyword)) {
            // 根据难度查询题目
            questionESPage = questionRepository.findQuestionByDifficulty(difficulty, pageable);
        } else if (difficulty == null) {
            // 根据标题或内容查询题目
            questionESPage = questionRepository.findByTitleOrContent(keyword, keyword, pageable);
        } else {
            // 根据标题、内容和难度查询题目
            questionESPage = questionRepository.findByTitleOrContentAndDifficulty(keyword, keyword, difficulty, pageable);
        }

        // 获取查询结果的总条数
        long total = questionESPage.getTotalElements();
        if (total <= 0) {
            return TableDataInfo.empty();
        }

        // 将 Elasticsearch 文档转换为 VO 对象
        List<QuestionES> questionESList = questionESPage.getContent();
        List<QuestionVO> questionVOList = BeanUtil.copyToList(questionESList, QuestionVO.class);

        // 返回分页数据
        return TableDataInfo.success(questionVOList, total);
    }

    /**
     * 查询热门题目列表。
     *
     * @return 热门题目列表
     */
    @Override
    public List<QuestionVO> hotList() {
        // 获取热门题目列表的大小
        Long total = questionCacheManager.getHostListSize();
        List<Long> hotQuestionIdList;
        if (total == null || total <= 0) {
            // 如果缓存中没有数据，从数据库中查询并刷新缓存
            PageHelper.startPage(Constants.HOST_QUESTION_LIST_START, Constants.HOST_QUESTION_LIST_END);
            hotQuestionIdList = userSubmitMapper.selectHostQuestionList();
            questionCacheManager.refreshHotQuestionList(hotQuestionIdList);
        } else {
            // 从缓存中获取热门题目列表
            hotQuestionIdList = questionCacheManager.getHostList();
        }

        // 组装热门题目VO列表
        return assembleQuestionVOList(hotQuestionIdList);
    }

    /**
     * 查询题目详情。
     *
     * @param questionId 题目ID
     * @return 题目详情
     */
    @Override
    public QuestionDetailVO detail(Long questionId) {
        // 从 Elasticsearch 中查询题目详情
        QuestionES questionES = questionRepository.findById(questionId).orElse(null);
        QuestionDetailVO questionDetailVO = new QuestionDetailVO();
        if (questionES != null) {
            // 如果 Elasticsearch 中有数据，直接返回
            BeanUtil.copyProperties(questionES, questionDetailVO);
            return questionDetailVO;
        }

        // 如果 Elasticsearch 中没有数据，从数据库中查询
        Question question = questionMapper.selectById(questionId);
        if (question == null) {
            return null;
        }

        // 刷新 Elasticsearch 数据
        refreshQuestion();

        // 将数据库中的题目数据转换为 VO 对象
        BeanUtil.copyProperties(question, questionDetailVO);
        return questionDetailVO;
    }

    /**
     * 获取当前题目的上一个题目ID。
     *
     * @param questionId 当前题目ID
     * @return 上一个题目ID
     */
    @Override
    public String preQuestion(Long questionId) {
        // 检查缓存中是否有数据，如果没有则刷新缓存
        Long listSize = questionCacheManager.getListSize();
        if (listSize == null || listSize <= 0) {
            questionCacheManager.refreshCache();
        }

        // 获取上一个题目ID
        return questionCacheManager.preQuestion(questionId).toString();
    }

    /**
     * 获取当前题目的下一个题目ID。
     *
     * @param questionId 当前题目ID
     * @return 下一个题目ID
     */
    @Override
    public String nextQuestion(Long questionId) {
        // 检查缓存中是否有数据，如果没有则刷新缓存
        Long listSize = questionCacheManager.getListSize();
        if (listSize == null || listSize <= 0) {
            questionCacheManager.refreshCache();
        }

        // 获取下一个题目ID
        return questionCacheManager.nextQuestion(questionId).toString();
    }

    /**
     * 刷新 Elasticsearch 中的题目数据。
     */
    private void refreshQuestion() {
        // 从数据库中查询所有题目
        List<Question> questionList = questionMapper.selectList(new LambdaQueryWrapper<Question>());
        if (CollectionUtil.isEmpty(questionList)) {
            return;
        }

        // 将数据库中的题目数据转换为 Elasticsearch 文档
        List<QuestionES> questionESList = BeanUtil.copyToList(questionList, QuestionES.class);

        // 将数据保存到 Elasticsearch 中
        questionRepository.saveAll(questionESList);
    }

    /**
     * 组装题目VO列表。
     *
     * @param hotQuestionIdList 热门题目ID列表
     * @return 题目VO列表
     */
    private List<QuestionVO> assembleQuestionVOList(List<Long> hotQuestionIdList) {
        if (CollectionUtil.isEmpty(hotQuestionIdList)) {
            return new ArrayList<>();
        }

        // 遍历热门题目ID列表，组装题目VO列表
        List<QuestionVO> resultList = new ArrayList<>();
        for (Long questionId : hotQuestionIdList) {
            QuestionVO questionVO = new QuestionVO();
            QuestionDetailVO detail = detail(questionId);
            questionVO.setTitle(detail.getTitle());
            resultList.add(questionVO);
        }
        return resultList;
    }
}
